Adding relationship to models