It seems to work - List comprehension + vectors and epsilon-greedy