Add the code from session 2.