prog1.py updated

This will train upon 70%data and will predict output for remaining 30%.