small edits in main