prog1.py updated

This will train upon 70% of the total data and will predict output for remaining 30%