import changes

just changed few lines of code