Refactor game screen/state into struct