diff --git a/README.md b/README.md
new file mode 100644
index 0000000..42ea0f6
--- /dev/null
+++ b/README.md
@@ -0,0 +1,8 @@
+## TODO
+
+- [ ] Get `torch.export` pipeline working.
+- [ ] Train tetris
+- [ ] Port `allegro-jax`
+
+
+
diff --git a/model.py b/model.py
index e9a6b61..7f4416a 100644
--- a/model.py
+++ b/model.py
@@ -140,4 +140,4 @@ def forward(self,
                                     torch.zeros(num_nodes, dtype=torch.int64),
                                     dim=0,
                                     dim_size=1)
-        return self.readout_mlp(graph_globals)
\ No newline at end of file
+        return self.readout_mlp(graph_globals)