Add assignment expression support and unit tests