Small modifications to session class
Updates