Update the code setup for easy addition.