changes to nklike fitting additional bandit