Create main.py

After Stefan's work. Check readme for more details.