Added support for while loops in 1 case.