One step closer, fixed a bug in computing U_1