Implemented AXPY in a generic way as an example