python task 2