Improve whitespace warnings for: else, return, while