Small fixes this time around. Just some quick bug fixes
TL; DR
Bug fixes
*** FIXED ***
- Predicted datasets that were dropping columns now include the full original input dataset