Minor changes to UI;