Modified main for evaluation. I'll fix it later