Fix graphing