Code and read me updated