Better documentation

Improved inline docs