Fixed a tiny bug in the main function. Was previosuly extracting the wrong argument.