added argument parsing to choose between visulaizer and network driver