Working on #assignment 2 :-)