Added unit tests for the new functionality