1. Add forget fucntion, but not complete