Added method to generate plots using a schematic