Add new function
Change import to test