Adds screenshot functionality