Added a main wrapper, so it can be imported