minor changes after review