Lab2 python