minor tweaks based on review