Make this a function and call it at the end of the file.