Add while statement