need to add update and handling of relationship