Correct pesky missing brackets
Style improvements