Change main to script