Kind of working now. Slow and inaccurate. Need more training data. Perhaps removing surf as well.