Code update

Readme edit