add gender prediction code