Re-wrote a bunch of the functions and added probability of child
friendly property feature
Also changed way train/dev/test sets are partitioned