(#4) Fix invaid JSON created by main.py