Add division code test