Added in ability to segment train file to perform tests outside of submitting to kaggle.