python exercise for CodingWithMe