Code i did for the interview