- Updated comments in main class