Wasn't saving full feature matrix...