Adding a README with an explanation plus some minor improvements to track.py and notebook