final readme update + slight code tweak