Allow challenge four to accept command line arguments