Update code for session 2
Add session 3 code