added tests for the while stmt