Add test to coverage
Add some tests for rest layer