Add session tests