added classes to contain node relationship