Assignment 2 working