The easiest option is just to use your OPT to run and build your business after you graduate and then move to O-1 or rely on STEM OPT to continue to run your company or work for another.