Modifing Main class