import gravity 
from gravity import Gravity, Production, Attraction, Doubly
import dispersion
import utils
import vec_SA
import count_model

import gravity 
from .gravity import Gravity, Production, Attraction, Doubly
import dispersion
import utils
import vec_SA
import count_model

update imports in init