1 files changed, 3 insertions, 2 deletions
diff --git a/include/RoundedRectangle.hpp b/include/RoundedRectangle.hpp
index f72a083..96e57fd 100644
--- a/include/RoundedRectangle.hpp
+++ b/include/RoundedRectangle.hpp
@@ -4,7 +4,7 @@
 #include <SFML/Graphics/Vertex.hpp>
 
 namespace sf {
-    class RenderWindow;
+    class RenderTarget;
     class Shader;
 }
 
@@ -14,9 +14,10 @@ namespace QuickMedia {
         RoundedRectangle(sf::Vector2f size, float radius, sf::Color color, sf::Shader *rounded_rectangle_shader);
         void set_position(sf::Vector2f pos);
         void set_size(sf::Vector2f size);
+        void set_color(sf::Color color);
         sf::Vector2f get_position() const;
         sf::Vector2f get_size() const;
-        void draw(sf::RenderWindow &window);
+        void draw(sf::RenderTarget &target);
     private:
         float radius;
         sf::Vector2f pos;