Updated MainTest for better line coverage~