state dict formatting