Add a main because a script doesn't work