Update to work with latest github UI changes

Should work with both unified and split diff views