Updated the way code blocks get rendered using schematics