Changed setup with random.sample