Trifacta interview question

(2nd interview) Test java skills. You are given a basic backbone program with 2 objects - a Library and a Book. Write functions to add, remove, rent, return books. Write Unit tests for the new functionality. What would you change if you had more time?