Fixing curly braces in integration code part 2