Minor ui update