Added tests for CI