updating some code