PEP8 fixes and rearranged images in plot