Updated task4.py

Pointed to right data