updated code

buxfix