Add test routing code