Add testing for custom routes