Improved UI
Bug fixes. Actually loads graph.