Made the inputs for the main function cleaner looking