spacing and small code tweak