Add matplotlib ax as plot argument

[meyer-nils]

This should enable plotting in an existing figure rather than creating a new one each time.

• Add argument
• Update plasticity examples with animation