If you want to reduce the number of lines here, you can use ```code.extend```