Create main.py

Still using the old ai.py. The new ai.py is still not working yet.