Added tests to example