Minor refactor to make code more readable
Make the nextGeneraton call actually call tick on the environment, returning the next state of the game