add : fix some bugs