Update server.py with feature_map return