Added answer for problem 2.4.
Added/Fixed comments.