minor changes and added visualization stuff"