Add automatic color coding to functionality