just added to the function