Update code for changes in react-router