#8 Some fixes
Added docs #8