Add needed braces