Minor updates to UI to add location