minor fixes after review