updated readme
updated code