Add some missing braces to make code cleaner.