Add flash() on checkout-address partial