update javascript to add colors