Added tests for the project