adjusted the code in the readme