Add code for the editor