HW 2- task 2. Completed. Tests were added.