Assignement 2 done