add code into code tests