Add comments in Main class