Edits to treadmill logic