Peer Assessment 1: corrected code