updated per @mr-c review