fixing little stuff