revised get_relation function to return dict keyed by sentence IDs with values a list of heads that match the given relation in that particular sentence