corrections to show reconstruction script for transformer dataset