Final version of assignment 4
Update final version of assignment 4