fix documentation
Readme and documentation fix
refactor problem 3