fix main so it runs