Adding documentation
Added Unit Test