fix bug and change example