Better grid, key handling moved to side functions
Game state handling separated from main game function, same with drawing - needs to be polished a bit