can predict properly, added feature names to traces