Added unit tests to example