Adding simple logic.
Added comments and logic change.
Merge branch 'master' of https://github.com/prasannabe2004/PythonLearning