diff --git a/README.md b/README.md
index 6e6a9b1..796a870 100644
--- a/README.md
+++ b/README.md
@@ -30,13 +30,13 @@ pip install mathgenerator
 Here is an example of how you would generate an addition problem:
 
 ```python
-from mathgenerator import mathgen
+import mathgenerator
 
 #generate an addition problem
-problem, solution = mathgen.addition()
+problem, solution = mathgenerator.addition()
 
 #another way to generate an addition problem using genById()
-problem, solution = mathgen.genById(0)
+problem, solution = mathgenerator.genById(0)
 ```
 <!--
 ### Creating a worksheet
@@ -55,8 +55,8 @@ ws.write_pdf()
 This creates the pdf `ws.pdf` in your current directory
 -->
 Problem/solution pairs are generated with either:
-* `mathgen.<generator_name>()` - generates a problem, solution set from the given generator name.
-* `mathgen.genById(id)` - generates a problem, solution set with generator id provided by the `id` parameter
+* `mathgenerator.<generator_name>()` - generates a problem, solution set from the given generator name.
+* `mathgenerator.genById(id)` - generates a problem, solution set with generator id provided by the `id` parameter
 
 <!--
 #### `format` kwarg
diff --git a/mathgenerator/__init__.py b/mathgenerator/__init__.py
index c726148..57eca79 100644
--- a/mathgenerator/__init__.py
+++ b/mathgenerator/__init__.py
@@ -4,7 +4,6 @@ from .generator import getGenList
 genList = getGenList()
 
 
-# || Non-generator Functions
 def genById(id, *args, **kwargs):
     generator = genList[id][2]
     return (generator(*args, **kwargs))