Add more unit tests

With the addition of XCTest today, we can focus on adding more unit tests to this project to aid in contributions and future pull requests.