Fix: correct main method signature