Updated assembly test code and visualization.