TODO fix JSON parsing