ENH: Updated the UI to reflect the latest collections api