Fixed Task 3 after comments.

Fixed Task 3 after comments.