task_01.py
part 1 of task 2