Refactoring
Added option for predicting the results after 1 or 2 sets
Added main run flow