Make graphing conditional