Add game states to update() and render()