For now, return the updated code.

Make sure to type return code.