Minor ui updates