1 files changed, 5 insertions, 0 deletions
diff --git a/TypeVar.hs b/TypeVar.hs
index 049a9e2..904eab7 100644
--- a/TypeVar.hs
+++ b/TypeVar.hs
@@ -63,3 +63,8 @@ mkEqualsType ty = typeFunc (AType [] (TypeOp (Name [] "bool"))) (typeFunc ty ty)
 
 typeFunc :: Type -> Type -> Type
 typeFunc ty1 ty2 = AType [ty1,ty2] (TypeOp (Name [] "->"))
+
+
+typeVarsInType :: Type -> Set Type
+typeVarsInType (TypeVar t) = Set.singleton (TypeVar t)
+typeVarsInType (AType list _) = unions . (map typeVarsInType) $ list