Fix image name + pep8