Style updates

Minor updates to match new Shopify UI