small changes after review