figuring out how to visualize the learning