minor updates based on review comments