Missing some curly brackets so adding them back in.