cleaned up plot a bit