Add custom messages and color support