added readme, and made some changes to the code?