Fixed main() method