the interview problem they gave me