Add parsing for while
Allow more whitespace