add new code in ch02