Updated styling logic