Use a better way to change the code