Very cool, thanks for sharing. How did you train it? Just manually labeling the data?