Added tests to the module