fixed the code for session 6