Adding documentation
Added a test