Assignment 2 coursework