separated class from test code