Get plotting working. Plots don't quite look right