Add session tests and functionality to session object