added training. Still need to implement html extraction + prediction
output