Add option to count marshmallows after sleeping