Added more functionality to session