Added new code for ADMM