Fixed colorizing function

It now colorizes correctly without any errors, it can be customized but
it is still under beta stage.