minor fixes
added plotting