Added small change to DataHandler.py.