Update some code

Update some code