Played a bit with main class