Improve Python image plotting

add plot labels.