Documentation added
Improved tests