Added error checking for if no argument is given.