Add getBooks function