code improve
Merge pull request #1 from sudhanshuptl/master

code improve