Enhancement #9 working

Enhancement #9 will work now. small change has been made to work.
Merge pull request #5 from karthikeyavecham/master

Enhancement #9 working