split code into separate modules, import from __init__