Adding some changes to visualize better.