Move game state handling into separate functions

[ReikoR]

Right now the states are handled in a long if-else statement. It would be better to put code for each state into separate function.