Assignment Week5, first iteration