did some minor changes
Merge pull request #1 from dilini-mampitiya/master

did some minor changes