small code tweak and readme update