Rewriting some vector assignments using axpy