Se puede hacer con algunas pequeñas modificaciones.
En lugar de agregar un listener a los eventos del mouse al document, tienes que agregar el listener al div que contiene la imagen. En lugar de usar un <img /> usar CSS (background-image) para mostrar la imagen y después cambiar el script para que los cuadros sean añadidos no al tag BODY sino al div que contiene la imagen.