Code simplification using pick()