add main.py and lib.py