distance_to_action needs to be rewritten to accomodate the bigger states