Add graphing support