Convert to XCTestCase
Add new files