Add more citations
remove some legacy code
add more comments
Expand `TRAIN_FOR_METRIC` section