need to add command line arguments