From 2e7859f068d9c936f5ae1f182001b51fbe124786 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Fri, 31 May 2024 13:06:50 +0200
Subject: [PATCH] Fixed README

---
 README.md | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 38c5fdc..8e20693 100644
--- a/README.md
+++ b/README.md
@@ -23,11 +23,10 @@ Fancy RL provides two main components:
 1. **Ready-to-use Classes for PPO / TRPL**: These classes allow you to quickly get started with reinforcement learning algorithms, enjoying the performance and hackability that comes with using TorchRL.
 
    ```python
-   from ppo import PPO
-   import gymnasium as gym
+   from fancy_rl import PPO
+
+   ppo = PPO("CartPole-v1")
 
-   env_spec = "CartPole-v1"
-   ppo = PPO(env_spec)
    ppo.train()
    ```