Add f1 score
Fix error in main.py