Modify theMainClass