It works! The f1 score shows 63.7% on the test set