Removed direct matplotlib calls from Session 5