Make the (a*x+b)^2 expansion work

Tests added. Benchmarks seem unaffected.