here are updates
updated code