Added tests to the package