Adding braces for if-else