Updated styling UI