map visalization works, TODO: make visualization of Q-matrix somehow?