Fixed some small errors with plotting and naming oops!