small changes that I forgot