docs fix, little refactor