update readme

README.md

@@ -87,3 +87,19 @@ The [`doc`](./doc) folder contains documentation about the rest of the code base
 * `doc/sokoban.md`: We include a small example to play a Sokoban level from the training data interactively.
 * `doc/mongodb.md`: Provides an overview of the included datasets with download links.
 * `doc/checkpoint_index.csv`: Lists all included checkpoints with download links.
+
+## License
+See the [LICENSE](./LICENSE) file for details about the license under which this code is made available.
+
+## Citation
+If you find this repository useful in your research, please consider giving a star :star: and a citation
+```
+@misc{lehnert2024beyondastar,
+      title={Beyond A*: Better Planning with Transformers via Search Dynamics Bootstrapping}, 
+      author={Lucas Lehnert and Sainbayar Sukhbaatar and DiJia Su and Qinqing Zheng and Paul Mcvay and Michael Rabbat and Yuandong Tian},
+      year={2024},
+      eprint={2402.14083},
+      archivePrefix={arXiv},
+      primaryClass={cs.AI}
+}
+```