Setup tests
Add class tests