Update outdated code