Finished up programming the function with the return.