Add main.py (for @Biinary )