Assignment 2 part 2 working?