changed main and made it compile