It's better to add
```python
if __name__ == "__main__":
```