HW 5 - task 1. Mistakes fixed, extra-test added