mainClass re-edit

Main class code modified. Imports added