diff --git a/eeg_eyetracking_parser/__init__.py b/eeg_eyetracking_parser/__init__.py
index a0af988..e55965f 100644
--- a/eeg_eyetracking_parser/__init__.py
+++ b/eeg_eyetracking_parser/__init__.py
@@ -8,6 +8,6 @@
tfr_to_surface
-__version__ = '0.15.0'
+__version__ = '0.15.1'
logger = logging.getLogger('eeg_eyetracking_parser')
logger.info(f'eeg_eyetracking_parser {__version__}')
diff --git a/readme.md b/readme.md
index cedeadc..a5c2a46 100644
--- a/readme.md
+++ b/readme.md
@@ -409,7 +409,7 @@ values between 128 and 255 (inclusive).
-## braindecode\_utils.decode\_subject_(read\_subject\_kwargs, factors, epochs\_kwargs, trigger, epochs\_query='practice == "no"', epochs=4, window\_size=200, window\_stride=1, n\_fold=4, crossdecode\_read\_subject\_kwargs=None, crossdecode\_factors=None, patch\_data\_func=None, read\_subject\_func=None, cuda=True)_
+## braindecode\_utils.decode\_subject_(read\_subject\_kwargs, factors, epochs\_kwargs, trigger, epochs\_query='practice == "no"', epochs=4, window\_size=200, window\_stride=1, n\_fold=4, crossdecode\_read\_subject\_kwargs=None, crossdecode\_factors=None, patch\_data\_func=None, read\_subject\_func=None, cuda=True, balance=True)_
The main entry point for decoding a subject's data.
@@ -494,6 +494,12 @@ The main entry point for decoding a subject's data.
If True, cuda will be used for GPU processing if it is available. If
False, cuda won't be used, not even when it is available.
+* **balance: bool, optional**
+
+ Makes sure that a dataset contains an equal number of observations for
+ each label by randomly duplicating observations from labels that have
+ too few observations.
+
### Returns
* **_DataMatrix_**