minor code tweak + update documentation