Add test for rest