Add python beat tracking code