add minor changes to the code