Add coloring definition for function with curly bracket on same line