Add support for command line argument

Checks for command line argument, otherwise prompts the user to enter one.