few changes in code