Assignment 4 finished