Update testing framework to XCTest