add screenshot and some minor tweaks
add support using a normal argument