diff --git a/src/Author.java b/src/Author.java
index 2057127..200d46d 100644
--- a/src/Author.java
+++ b/src/Author.java
@@ -5,11 +5,11 @@
 import java.util.List;
 
 public class Author extends Person {
-    private List books;
+    private final ArrayList<String> books;
 
     public Author(String firstName, String lastName) {
         super(firstName, lastName);
-        books = new ArrayList();
+        books = new ArrayList<String>();
     }
 
     /**
@@ -28,7 +28,7 @@ public void addBook(String book) {
         books.add(book);
     }
 
-    @Override
+
     public String sortName() {
         return String.format("%s, %s", lastName, firstName);
     }
diff --git a/src/Main.java b/src/Main.java
index 08139ff..c360922 100644
--- a/src/Main.java
+++ b/src/Main.java
@@ -5,7 +5,7 @@ public static void main(String[] args) {
         author.addBook("Practical Object-Oriented Design in Ruby");
         author.addBook("99 Bottles of OOP");
 
-        for (String book: author.getBooks()) {
+        for (String book: author.publishedBooks()) {
             System.out.println(book);
         }
     }