Changing the main class.
Changing main around