Minor ui adjustments