Agent wants to select the best action, but encoding is in distance, not direction. This needs fixing