Fixed color coding change for new feature!