Added proper main() method