fixed ADMM code