bug fix & add export