Changes to get the code from the AI server