add color coding