updated code

here it is