Tried some changes which appear to have broken EVERYTHING!
Fixed a minor visualization bug.