Final Project Submission with a minor code change