diff --git a/src/main.py b/src/main.py
index 2dd9758..fe2bbd3 100644
--- a/src/main.py
+++ b/src/main.py
@@ -81,6 +81,8 @@ def main(opt):
         phi, num_authors=A.size, pad_index=pad_index, pad_length=opt.pad, device=device
     )
 
+    print(cls)
+
     if opt.name == 'auto':
         method = f'{phi.__class__.__name__}_alpha{opt.alpha}'
     else:
diff --git a/src/model/classifiers.py b/src/model/classifiers.py
index ebdb57f..93948e4 100644
--- a/src/model/classifiers.py
+++ b/src/model/classifiers.py
@@ -13,8 +13,11 @@ class AuthorshipAttributionClassifier(nn.Module):
     def __init__(self, projector, num_authors, pad_index, pad_length=500, device='cpu'):
         super(AuthorshipAttributionClassifier, self).__init__()
         self.projector = projector.to(device)
+        #self.ff = FFProjection(input_size=projector.space_dimensions(),
+        #                       hidden_sizes=[1024],
+        #                       output_size=num_authors).to(device)
         self.ff = FFProjection(input_size=projector.space_dimensions(),
-                               hidden_sizes=[1024],
+                               hidden_sizes=[],
                                output_size=num_authors).to(device)
         self.padder = Padding(pad_index=pad_index, max_length=pad_length, dynamic=True, pad_at_end=False, device=device)
         self.device = device