added visualization option
added an argument
added violation colors