Adds the missing imports from pull request #7.
Moves checkConnection to DataController as suggested in #7.