Add support for while statement.