bug fix & add reset