Edit: Setup color coding