General UI layout improvements