add import random.sample