slight refactor of code and update of the read me file.