Fixing the imports in server.py