clean up. Add command line arg and help