Perfeito. Apenas sugiro que ao invés de usar screen.width e screen.height use document.documentElements.clientWidth e document.documentElements.clientHeight, já que screen.* pega a resolução de tela e não a resolução da janela do navegador. Caso não funcione dentro da função tente passar como parâmetro ao chamar a função.