Changed to accept argument from commandline