Python Write Json Output