Code style change
Add assert header
Add asserts in session