Modified data.py for FIFE