Updated controller code with latest KFStatsX changes