Working small styling updates