handle command-line args better