Removed relative import as long as this is a Python 2 module only