adding more unit tests