Improve handling of assignment binary operators.

Fixes simple cases such as result = x*x;