upadating to cater a missing import.
Merge pull request #30 from abeykoon/refectoring-stage1

upadating to cater a missing import.