Homework 1 with command line arguments