Particle In Cell code written in Python