Added code

Changed the readme