Graphs are working but super hacky