Add extra curly braces to the while loop