Added new function to models