Finish unit tests for books