%%
%% The LaTeX Companion, 3ed
%%
%% Example 11-7-23 on page II-208 in "Color in formulas".
%%
%% Copyright (C) 2022 Frank Mittelbach
%%
%% It may be distributed and/or modified under the conditions
%% of the LaTeX Project Public License, either version 1.3c
%% of this license or (at your option) any later version.
%%
%% See https://www.latex-project.org/lppl.txt for details.
%%

\documentclass{tlc3examargin}
\pagestyle{empty}
\setcounter{page}{6}
\setlength\textwidth{135.0pt}

 % Shorten the vertical size of the example ...
 \AtBeginDocument{%
   \setlength\abovedisplayskip{0.2\abovedisplayskip}%
   \setlength\belowdisplayskip{.1\belowdisplayskip}%
 }

%StartShownPreambleCommands
\usepackage{amsmath,color}
%StopShownPreambleCommands

\begin{document}
\begin{align*}
 (a+b)^2 &= a^2 \color{blue}+2ab\color{black} + b^2 \\
 (a-b)^2 &= a^2 \mathcolor{blue}{-2ab} + b^2        \\
\intertext{But don't do this:}
 (a-b)^2 &= a^2 \textcolor{blue}{-2ab}+ b^2 \end{align*}
\end{document}