small fix
Merge pull request #5 from vikdatta/master

Small fixes