minor changes from review