readd screenshot functionality