Add test with rake spec