Add docstrings in MUX and update README