minor changes based on review