Slight changes

To game logic