Updated class 2 homework to use ARGV