temp.ipynb is the practice file main.ipynb is the the file where I tested the main code main.py is the final file