Adding correct method to plotting