Put while statement back in for test.