Ability to change main class