Session updates

Adding the unit tests and session logic to the repo.