Update and rename main.py to add_func.py

only keep add function