the code related to my question!