Refactoring : Divide GameStage class