Moving code to their respective classes.
This was done to make it easier and more organised for
everyone to work with. We are now also using ScreenManager
window size.