added screenshot functionality