Add colorize to docs
Clean up