Started working on the while statement