Write assignment 2