Import main

I forgot to import main because I'm a dunce.