adding more unit tests

adding unit tests