Add code and update readme
Added code and update readme