Made it possible to use command line arguments