Huskers working on fixing discipline