Update pick script

Change-Id: Ia8f5a3682fdc2dcba2502a11af3ccdbb57ef861d