Add to_json function