add classes needed for testing