code tweak fixes