Add functionality to router