Add missing docblocks
Improve documentation