some documentation and move plot fns to plot module
speeding things up