Revert change to tpms.py (fix broken test).
Fix script structure of calculate_reads_for_depth.