Updated the algo