Fixed more things

Visualization is cool guy