Session 3 solution fix