small corrections to CODE generator