Updated color detection