Added flexibility to orbits.py to handle N bodies