added plot stuff