Tidying piece image code slightly