Refactored boid.py