Minor tidyup and fixed broken image