Add code to the functions