change main.py

Implementation tentatively.
But this doesn't work well ...