Revised the answer to the exercises

They can be refined!