diff --git a/user_IO.py b/user_IO.py
index e2e62a6..c53e577 100644
--- a/user_IO.py
+++ b/user_IO.py
@@ -12,5 +12,14 @@ def userinput():
             subject = input("Which subject do you want to add? \n")
             return "input", subject
         elif todo == "category":
-            subject = input("Where do you want to add the category? \n")
-            category = input("Which category do you want ot add? \n")
\ No newline at end of file
+            subject = input("To which subject do you want to add the category? \n")
+            category = input("Which category do you want to add? \n")
+            rating = float(input("Rating? \n"))
+            return "input", subject, category, rating
+        elif todo == "grade":
+            subject = input("To which subject do you want to add the grade? \n")
+            category = input("To which category do you want to add it? \n")
+            grade = input("Grade? \n")
+            gradename = input("What's the name of the grade? \n")
+            return "input", subject, category, grade, gradename
+