code updated
updated code