Fixed some weird indentation problems in the main function