HW - part2 (done).