Small documentation update
Added unit tests