add solution to hw 01 task 2