update for new pkl directory