Master notebook for generating training and test sets