Plots look like they are correct now, probably should refactor.