Extract showing/hiding of screenshot button into own method