diff --git a/README.md b/README.md
index 68a1a2b..038399b 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,10 @@
-# Fancy RL
+<h1 align="center">
+  <br>
+  <img src='./fancy_rl.svg' width="250px">
+  <br><br>
+  <b>Fancy RL</b>
+  <br><br>
+</h1>
 
 Fancy RL is a minimalistic and efficient implementation of Proximal Policy Optimization (PPO) and Trust Region Policy Layers (TRPL) using primitives from [torchrl](https://pypi.org/project/torchrl/). Future plans include implementing Soft Actor-Critic (SAC). This library focuses on providing clean and understandable code while leveraging the powerful functionalities of torchrl.
 We provide optional integration with wandb.
diff --git a/fancy_rl.svg b/fancy_rl.svg
new file mode 100644
index 0000000..751041c
--- /dev/null
+++ b/fancy_rl.svg
@@ -0,0 +1,139 @@
+<?xml version="1.0" encoding="UTF-8" standalone="no"?>
+<!-- Created with Inkscape (http://www.inkscape.org/) -->
+
+<svg
+   width="209.35104mm"
+   height="207.26176mm"
+   viewBox="0 0 209.35104 207.26176"
+   version="1.1"
+   id="svg1"
+   xml:space="preserve"
+   inkscape:version="1.3.2 (091e20ef0f, 2023-11-25)"
+   sodipodi:docname="fancy_rl.svg"
+   xmlns:inkscape="http://www.inkscape.org/namespaces/inkscape"
+   xmlns:sodipodi="http://sodipodi.sourceforge.net/DTD/sodipodi-0.dtd"
+   xmlns="http://www.w3.org/2000/svg"
+   xmlns:svg="http://www.w3.org/2000/svg"><sodipodi:namedview
+     id="namedview1"
+     pagecolor="#ffffff"
+     bordercolor="#999999"
+     borderopacity="1"
+     inkscape:showpageshadow="0"
+     inkscape:pageopacity="0"
+     inkscape:pagecheckerboard="0"
+     inkscape:deskcolor="#d1d1d1"
+     inkscape:document-units="mm"
+     inkscape:zoom="0.7622131"
+     inkscape:cx="172.52393"
+     inkscape:cy="565.45866"
+     inkscape:window-width="3440"
+     inkscape:window-height="1403"
+     inkscape:window-x="0"
+     inkscape:window-y="0"
+     inkscape:window-maximized="1"
+     inkscape:current-layer="layer1-36" /><defs
+     id="defs1"><linearGradient
+       id="linearGradient3331"
+       inkscape:swatch="solid"><stop
+         style="stop-color:#000000;stop-opacity:1;"
+         offset="0"
+         id="stop3329" /></linearGradient></defs><g
+     inkscape:label="Layer 1"
+     inkscape:groupmode="layer"
+     id="layer1"
+     transform="translate(-0.25629321,-44.869116)"><g
+       inkscape:label="Layer 1"
+       id="layer1-36"
+       transform="translate(-0.07908684,-44.675806)"><path
+         id="path243"
+         style="fill:#1b9783;fill-opacity:1;stroke-width:1.82698"
+         d="M 105.01116,89.544922 C 47.200277,89.544813 0.33526959,135.94201 0.33538005,193.17581 0.33527406,250.40961 47.20028,296.8068 105.01116,296.80669 162.82184,296.80652 209.68654,250.40941 209.68643,193.17581 209.68654,135.9422 162.82184,89.545091 105.01116,89.544922 Z"
+         sodipodi:nodetypes="ccccc" /><g
+         id="g8676"
+         transform="matrix(0.35277778,0,0,0.35277778,93.730168,-46.079548)"><path
+           d="m 210.96344,516.0861 12.37693,-4.7124 -12.37693,-4.71237 v -0.002 c -2.44481,-0.93931 -4.69468,-3.36861 -6.45332,-6.97435 -1.75974,-3.6034 -2.94651,-8.21411 -3.4035,-13.22394 l -2.29965,-25.36242 -2.29965,25.36242 h -9.7e-4 c -0.45329,5.01503 -1.63749,9.62822 -3.3973,13.23664 -1.76102,3.60599 -4.01242,6.03262 -6.45951,6.96165 l -12.37693,4.71238 12.39453,4.71498 c 2.44862,0.92368 4.70251,3.35031 6.46212,6.95631 1.76101,3.60862 2.94394,8.22414 3.3947,13.24199 l 2.29965,25.32299 2.29966,-25.32299 h 9.6e-4 c 0.44949,-5.01245 1.63115,-9.62555 3.38851,-13.23197 1.75593,-3.60861 4.0059,-6.03791 6.45071,-6.967 z"
+           id="path8600"
+           style="fill:#ffffff;fill-opacity:1;stroke-width:0.46663" /><path
+           d="m 241.05123,518.51515 c 0.30634,-3.04032 1.58367,-5.41746 3.21463,-5.98793 l 4.02723,-1.3888 -4.02723,-1.3888 c -1.63567,-0.55553 -2.91649,-2.94127 -3.21463,-5.98793 l -0.74559,-7.50158 -0.74558,7.50158 c -0.30634,3.03817 -1.58251,5.41746 -3.21463,5.98793 l -4.02723,1.3888 4.02723,1.3888 c 1.62528,0.59427 2.89569,2.96067 3.21463,5.98793 l 0.74558,7.50158 z"
+           id="path8602-6-7"
+           style="fill:#ffffff;fill-opacity:1;stroke-width:0.403881" /><path
+           d="m 177.89263,495.09636 c 0.25962,-2.6322 1.34213,-4.69025 2.72434,-5.18414 l 3.413,-1.20237 -3.413,-1.20238 c -1.3862,-0.48096 -2.47167,-2.54645 -2.72434,-5.18414 l -0.63187,-6.4946 -0.63187,6.4946 c -0.25961,2.63034 -1.34114,4.69025 -2.72433,5.18414 l -3.413,1.20238 3.413,1.20237 c 1.37739,0.5145 2.45404,2.56324 2.72433,5.18414 l 0.63187,6.4946 z"
+           id="path8602-6-7-6"
+           style="fill:#ffffff;fill-opacity:1;stroke-width:0.345954" /><path
+           d="m 231.84066,571.8627 c 0.32831,-3.0209 1.18695,-5.79883 2.46229,-7.9706 1.27354,-2.17492 2.90247,-3.63925 4.67382,-4.1991 l 8.97453,-2.60256 -8.97453,-2.86707 c -1.76959,-0.55373 -3.39852,-2.00884 -4.67198,-4.17454 -1.27353,-2.1657 -2.134,-4.93741 -2.46408,-7.95247 l -1.73354,-15.52615 -1.65595,15.30485 c -0.33011,3.01474 -1.19055,5.78653 -2.46408,7.95247 -1.27355,2.16594 -2.90249,3.62083 -4.67199,4.17454 l -8.86969,3.08861 8.94728,2.82401 v -0.003 c 1.77139,0.56295 3.40033,2.02725 4.67383,4.19911 1.27533,2.17491 2.13399,4.95284 2.46409,7.97374 l 1.57661,15.04023 z"
+           id="path8604"
+           style="fill:#ffffff;fill-opacity:1;stroke-width:0.603057" /><path
+           d="m 162.84035,471.82125 c 0.48729,-2.70075 1.76171,-5.1843 3.65462,-7.12591 1.89023,-1.94444 4.30794,-3.25359 6.93705,-3.75411 l 13.32028,-2.32675 -13.32028,-2.56324 c -2.6265,-0.49504 -5.04422,-1.79595 -6.93432,-3.73215 -1.89022,-1.93619 -3.16735,-4.41416 -3.65728,-7.10971 l -2.57297,-13.88077 -2.45781,13.68293 c -0.48996,2.69525 -1.76706,5.1733 -3.65729,7.10971 -1.89023,1.93641 -4.30796,3.23711 -6.93431,3.73214 l -13.1647,2.7613 13.27985,2.52474 v -0.003 c 2.62916,0.5033 5.04689,1.81242 6.93705,3.75412 1.89291,1.94442 3.16735,4.42797 3.65729,7.12872 l 2.34007,13.44636 z"
+           id="path8604-3"
+           style="fill:#ffffff;fill-opacity:1;stroke-width:0.694679" /></g><g
+         id="g1"
+         transform="matrix(0.30355272,0,0,0.30355272,34.847653,121.29032)"
+         style="fill:#ffffff"><g
+           id="g5"
+           style="fill:#ffffff">
+	<path
+   d="M 151.245,222.446 C 148.054,237.039 135.036,248 119.5,248 c -4.142,0 -7.5,3.357 -7.5,7.5 0,4.143 3.358,7.5 7.5,7.5 23.774,0 43.522,-17.557 46.966,-40.386 14.556,-1.574 27.993,-8.06 38.395,-18.677 2.899,-2.959 2.85,-7.708 -0.109,-10.606 -2.958,-2.897 -7.707,-2.851 -10.606,0.108 C 184.947,202.829 172.643,208 159.5,208 132.757,208 111,186.243 111,159.5 c 0,-4.143 -3.358,-7.5 -7.5,-7.5 -4.142,0 -7.5,3.357 -7.5,7.5 0,32.215 24.119,58.884 55.245,62.946 z"
+   id="path1"
+   style="fill:#ffffff" />
+	<path
+   d="m 183,287.5 c 0,-4.143 -3.358,-7.5 -7.5,-7.5 -35.014,0 -63.5,28.486 -63.5,63.5 0,0.362 0.013,0.725 0.019,1.088 C 109.23,344.212 106.39,344 103.5,344 c -4.142,0 -7.5,3.357 -7.5,7.5 0,4.143 3.358,7.5 7.5,7.5 26.743,0 48.5,21.757 48.5,48.5 0,4.143 3.358,7.5 7.5,7.5 4.142,0 7.5,-3.357 7.5,-7.5 0,-26.611 -16.462,-49.437 -39.731,-58.867 -0.178,-1.699 -0.269,-3.418 -0.269,-5.133 0,-26.743 21.757,-48.5 48.5,-48.5 4.142,0 7.5,-3.357 7.5,-7.5 z"
+   id="path2"
+   style="fill:#ffffff" />
+	<path
+   d="m 439,223.5 c 0,-17.075 -6.82,-33.256 -18.875,-45.156 C 422.034,172.236 423,165.918 423,159.5 423,128.626 400.848,102.841 371.606,97.171 373.841,91.6 375,85.628 375,79.5 375,59.943 363.117,43.113 346.194,35.839 317.999,13.383 287.162,0 263.5,0 250.347,0 238.683,6.468 231.5,16.384 224.317,6.468 212.653,0 199.5,0 175.838,0 145.001,13.383 116.806,35.839 99.883,43.113 88,59.943 88,79.5 88,85.628 89.159,91.6 91.394,97.171 62.152,102.841 40,128.626 40,159.5 c 0,6.418 0.965,12.735 2.875,18.844 C 30.82,190.244 24,206.425 24,223.5 c 0,13.348 4.149,25.741 11.213,35.975 C 27.872,270.087 24,282.466 24,295.5 c 0,23.088 12.587,44.242 32.516,55.396 -0.343,2.852 -0.516,5.73 -0.516,8.604 0,31.144 20.315,58.679 49.79,68.063 12.821,21.942 36.175,35.437 61.71,35.437 27.995,0 52.269,-16.181 64,-39.674 11.731,23.493 36.005,39.674 64,39.674 25.535,0 48.889,-13.495 61.71,-35.437 29.475,-9.385 49.79,-36.92 49.79,-68.063 0,-2.874 -0.173,-5.752 -0.516,-8.604 C 426.413,339.742 439,318.588 439,295.5 439,282.466 435.128,270.087 427.787,259.475 434.851,249.241 439,236.848 439,223.5 Z M 167.5,448 c -21.029,0 -40.191,-11.594 -50.009,-30.256 -0.973,-1.849 -2.671,-3.208 -4.688,-3.751 C 88.19,407.369 71,384.961 71,359.5 c 0,-3.81 0.384,-7.626 1.141,-11.344 0.702,-3.447 -1.087,-6.92 -4.302,-8.35 C 50.32,332.018 39,314.626 39,295.5 39,286.801 41.256,278.486 45.561,271.121 56.757,280.992 71.436,287 87.5,287 91.642,287 95,283.643 95,279.5 95,275.357 91.642,272 87.5,272 60.757,272 39,250.243 39,223.5 c 0,-14.396 6.352,-27.964 17.428,-37.221 2.5,-2.09 3.365,-5.555 2.14,-8.574 C 56.2,171.869 55,165.744 55,159.5 55,132.757 76.757,111 103.5,111 c 26.743,0 48.5,21.757 48.5,48.5 0,4.143 3.358,7.5 7.5,7.5 4.142,0 7.5,-3.357 7.5,-7.5 C 167,125.858 140.698,98.257 107.579,96.145 104.577,91.127 103,85.421 103,79.5 c 0,-13.369 8.116,-24.875 19.678,-29.859 0.447,-0.133 0.885,-0.307 1.308,-0.527 C 127.568,47.752 131.447,47 135.5,47 c 12.557,0 23.767,7.021 29.256,18.325 1.81,3.727 6.298,5.281 10.023,3.47 3.726,-1.809 5.28,-6.296 3.47,-10.022 C 171.983,45.87 160.124,36.596 146.467,33.311 165.609,21.631 184.454,15 199.5,15 213.009,15 224,25.99 224,39.5 v 97.051 C 217.261,131.205 208.75,128 199.5,128 c -4.142,0 -7.5,3.357 -7.5,7.5 0,4.143 3.358,7.5 7.5,7.5 13.509,0 24.5,10.99 24.5,24.5 v 180.279 c -9.325,-12.031 -22.471,-21.111 -37.935,-25.266 -3.999,-1.071 -8.114,1.297 -9.189,5.297 -1.075,4.001 1.297,8.115 5.297,9.189 C 206.8,343.616 224,366.027 224,391.5 224,422.654 198.654,448 167.5,448 Z M 395.161,339.807 c -3.215,1.43 -5.004,4.902 -4.302,8.35 0.757,3.718 1.141,7.534 1.141,11.344 0,25.461 -17.19,47.869 -41.803,54.493 -2.017,0.543 -3.716,1.902 -4.688,3.751 C 335.691,436.406 316.529,448 295.5,448 264.346,448 239,422.654 239,391.5 c 0,-2.109 -0.098,-4.2 -0.281,-6.271 0.178,-0.641 0.281,-1.314 0.281,-2.012 V 135.5 c 0,-13.51 10.991,-24.5 24.5,-24.5 4.142,0 7.5,-3.357 7.5,-7.5 0,-4.143 -3.358,-7.5 -7.5,-7.5 -9.25,0 -17.761,3.205 -24.5,8.551 V 39.5 C 239,25.99 249.991,15 263.5,15 c 15.046,0 33.891,6.631 53.033,18.311 -13.657,3.284 -25.516,12.559 -31.782,25.462 -1.81,3.727 -0.256,8.214 3.47,10.022 3.726,1.81 8.213,0.257 10.023,-3.47 C 303.733,54.021 314.943,47 327.5,47 c 4.053,0 7.933,0.752 11.514,2.114 0.422,0.22 0.86,0.393 1.305,0.526 C 351.883,54.624 360,66.13 360,79.5 c 0,5.921 -1.577,11.627 -4.579,16.645 C 322.302,98.257 296,125.858 296,159.5 c 0,4.143 3.358,7.5 7.5,7.5 4.142,0 7.5,-3.357 7.5,-7.5 0,-26.743 21.757,-48.5 48.5,-48.5 26.743,0 48.5,21.757 48.5,48.5 0,6.244 -1.2,12.369 -3.567,18.205 -1.225,3.02 -0.36,6.484 2.14,8.574 11.075,9.257 17.427,22.825 17.427,37.221 0,26.743 -21.757,48.5 -48.5,48.5 -4.142,0 -7.5,3.357 -7.5,7.5 0,4.143 3.358,7.5 7.5,7.5 16.064,0 30.743,-6.008 41.939,-15.879 4.306,7.365 6.561,15.68 6.561,24.379 0,19.126 -11.32,36.518 -28.839,44.307 z"
+   id="path3"
+   style="fill:#ffffff" />
+	<path
+   d="M 359.5,240 C 343.964,240 330.946,229.039 327.755,214.446 358.881,210.384 383,183.715 383,151.5 c 0,-4.143 -3.358,-7.5 -7.5,-7.5 -4.142,0 -7.5,3.357 -7.5,7.5 0,26.743 -21.757,48.5 -48.5,48.5 -13.143,0 -25.447,-5.171 -34.646,-14.561 -2.898,-2.958 -7.647,-3.007 -10.606,-0.108 -2.959,2.899 -3.008,7.647 -0.109,10.606 10.402,10.617 23.839,17.103 38.395,18.677 3.444,22.829 23.192,40.386 46.966,40.386 4.142,0 7.5,-3.357 7.5,-7.5 0,-4.143 -3.358,-7.5 -7.5,-7.5 z"
+   id="path4"
+   style="fill:#ffffff" />
+	<path
+   d="m 335.5,328 c -2.89,0 -5.73,0.212 -8.519,0.588 0.006,-0.363 0.019,-0.726 0.019,-1.088 0,-35.014 -28.486,-63.5 -63.5,-63.5 -4.142,0 -7.5,3.357 -7.5,7.5 0,4.143 3.358,7.5 7.5,7.5 26.743,0 48.5,21.757 48.5,48.5 0,1.714 -0.091,3.434 -0.269,5.133 C 288.462,342.063 272,364.889 272,391.5 c 0,4.143 3.358,7.5 7.5,7.5 4.142,0 7.5,-3.357 7.5,-7.5 0,-26.743 21.757,-48.5 48.5,-48.5 4.142,0 7.5,-3.357 7.5,-7.5 0,-4.143 -3.358,-7.5 -7.5,-7.5 z"
+   id="path5"
+   style="fill:#ffffff" />
+</g><g
+           id="g6"
+           style="fill:#ffffff">
+</g><g
+           id="g7"
+           style="fill:#ffffff">
+</g><g
+           id="g8"
+           style="fill:#ffffff">
+</g><g
+           id="g9"
+           style="fill:#ffffff">
+</g><g
+           id="g10"
+           style="fill:#ffffff">
+</g><g
+           id="g11"
+           style="fill:#ffffff">
+</g><g
+           id="g12"
+           style="fill:#ffffff">
+</g><g
+           id="g13"
+           style="fill:#ffffff">
+</g><g
+           id="g14"
+           style="fill:#ffffff">
+</g><g
+           id="g15"
+           style="fill:#ffffff">
+</g><g
+           id="g16"
+           style="fill:#ffffff">
+</g><g
+           id="g17"
+           style="fill:#ffffff">
+</g><g
+           id="g18"
+           style="fill:#ffffff">
+</g><g
+           id="g19"
+           style="fill:#ffffff">
+</g><g
+           id="g20"
+           style="fill:#ffffff">
+</g></g></g></g></svg>