Exercício com cin e cout em C

Altere o programa feito para dizer se um número é primo, de maneira que ele utilize apenas os comandos cin e cout para entrada e saída de dados.

#include <iostream.h>
int primo(int numero);
void main(){
    int entrada;
    //printf("Digite um numero: ");
    cout << "Digite um numero ";
    //scanf("%d",&entrada);
    cin >> entrada;
    if(primo(entrada))
        //printf("\nNumero Primo");
        cout << "\nNumero Primo";
    else //printf("\nNumero nao Primo");
        cout << "\nNumero nao Primo";
    getch();
}
int primo(int numero){
    int i;
    for(i=2;i<numero;i++)
    {
        if ((numero % i) == 0)
            return 0;
    }
    return 1;
}