Added belongs functions to models