need to add relationships