did some work. added plot