Simple Particle filter implementation in Python