Added ability to choose correct and wrong colors

Signed-off-by: Tabor Kvasnicka <kamakwazee@gmail.com>