Fix color coding attempt1