Add os.path import in os module