minor changes in code