Day2 python mini project