added plot code, worked with simion on improvements