Fixed small bugs with the game