Adding game logic, WIP