Small edits to main file