Improve resizing and add better screenshot