add command line flag for fast training