Update this file for using the function