fix & add codes
fix a little error