Assignment 4.1 finished