Set the viewport properly