Updating some code