Small updates to the util, removing the prints and the tuple return