Parse the correct arguments in main