Add test for custom method