I made some changes to challenge #4