Update __init__.py to support midi input