Assignment 2 work