Update mtree.py

Fix python syntax

Closes #3