Fixing documentation

Adding / fixing documentation