Fixed main to build