Add final code after session and comments