Working on CG solver