Slight change to Week5Assignment