added answer to assignment 5.4 from lesson 2