Cleaned up flocking AI code a bit more