small improvement
small changes