Updating missing code