Modifying our algorithm