Added tests to the project