Small fixes for game