Fixed small error in main