Include shape fingerprint in train.py