Add run only to unittests
Fix class name
Use global session