Initial task 2 (some mistakes)