kata 4 python