from colorama import init, Fore, Style

init()


def error(text):
    print(Fore.RED + 'Error: %s' % text + Style.RESET_ALL)


def info(*args, **kwargs):
    print(*args, **kwargs)


def good(text):
    return Fore.GREEN + text + Style.RESET_ALL


def bad(text):
    return Fore.RED + text + Style.RESET_ALL

@@ -37,16 +37,17 @@
 le%0A%0A
+%0A
 init()%0A%0A
 %0Adef
@@ -42,16 +42,65 @@
 init()%0A%0A
+# TODO: dependency on colorama really necessary?%0A
 %0Adef err
@@ -169,17 +169,16 @@
 T_ALL)%0A%0A
-%0A
 def info
@@ -224,17 +224,16 @@
 wargs)%0A%0A
-%0A
 def good
@@ -288,17 +288,16 @@
 ET_ALL%0A%0A
-%0A
 def bad(
@@ -295,32 +295,32 @@
 %0Adef bad(text):%0A
-
     return Fore.
@@ -328,28 +328,93 @@
 ED + text + Style.RESET_ALL%0A
+%0Adef okay(text):%0A    return Fore.YELLOW + text + Style.RESET_ALL%0A

Add yellow output color