Add styling updates