Made a main wrapper for both parts of the project