cleaned up code, commented out visualization for now