Added tracking code!

 Minor bug fix in previous version. Removed the `if` condition.