Add "code" support