Python Interview question and solution