Small changes to style for better ui