Add preliminary unit tests