Added tests to the repo