add assignment 2 answer to the code.