Updating styles to match latest html class