Replaced write_feature_file with NDJWriter

- Used NDJWriter instead of write_feature_file where we were specifically writing .jsonlines files.