Add main script
Remove init.py