First exercise in assignment 1 completed