Add fixViewport function