Fixed return bug + changed payload to 2d dict