diff --git a/github_issue_summarization/notebooks/issue_summarization.py b/github_issue_summarization/notebooks/IssueSummarization.py
similarity index 90%
rename from github_issue_summarization/notebooks/issue_summarization.py
rename to github_issue_summarization/notebooks/IssueSummarization.py
index 2286d444..b992b2dd 100644
--- a/github_issue_summarization/notebooks/issue_summarization.py
+++ b/github_issue_summarization/notebooks/IssueSummarization.py
@@ -21,5 +21,5 @@ class IssueSummarization(object):
                                    decoder_preprocessor=title_pp,
                                    seq2seq_model=load_model('seq2seq_model_tutorial.h5'))
 
-  def predict(self, input_text):
+  def predict(self, input_text, feature_names): # pylint: disable=unused-argument
     return np.asarray([[self.model.generate_issue_title(body[0])[1]] for body in input_text])