New Readme, small changes in code