updating some code and the readme