Modified the UI design