Edited the solution part
Merge pull request #1 from vimuth/patch-1

Edited the solution part