Added command line arg for input and finished the assignment.