Updated to use XCTest rather than GHUnit