galaxy sim in python