Code for the Kahn Academy interview