Fixed a couple of silly bugs
The decision tree should now be fairly consistent.