still working on making plots nice