Add 'while' statement formatter