fixed getting features, previously only feature_dict was pickled