Small tweaks to UI
Update paper ref