Python task 2