Initial code from after the interview