- small changes based on review feedback