move import os just below the func def
Merge pull request #125 from ruturajv/master

move import os just below the func def