Start refactoring to classes
Refactor to classes
Add support for game restart