minor improvemeents