Still problem using cvxpy. Likely not to do with the reshaping of variables