Fixes some comments in main file for clarification