Fixes with remove books. Unit tests