fix type bug, cleanup image display