added imports and before to test class