Update some code