changed main to match what I need