Include this in the main function to use

You need this to run this in the main function. Sadly I didn't comment it out.