Fix initial code py3