minor changes in the code