Edits made to problem #3