Image viewer UI improvements