That's everything! When doing this on your own, definitely use sklearn to experiment but you'll want to write the logic that trains the tree on your own. You'd also want more data!