Need to add command line flags