Add ensure function
Add pick function