Python Interview Bit Question