added sys import and when no command line args, guides user to enter learn or predict