Slight bug fix. Graph works better now.