add relationship
add logic