Now have a working 2d mpm python code - next step - a bit of optimisation