Main class inserted
Changes tests of Main