fix bug and adding demo