This is a nit, but I'd prefer if you'd follow my style of only adding braces when necessary (or in an `else` chain when one of the branches needs braces)