File 5: Putting the code in a function