Modification

Added minor error handling to code when run without any arguments.