Minor styling and ui fixes