edited main.py. Now we have pictures :D