Both H1b and O1 work in this case with the limitation that in H1b you can't own majority share of the company, i.e. someone needs to be able to kick you out. I'm not aware of all the details but you should consult an immigration lawyer about this.