[shitou] - refactor code, then add methid to create custom menu