changed Main class