cleaned up imaging figure