Test program for C file writing

#include
#include
#include

int main()
{
FILE *fp;
char ch, filename[40];
float data[5];
int count;

puts("Enter 5 floating point numbers");

for( count = 0; count < 5; count++ )
scanf("%f", &data[count]);

fflush(stdin);

printf("Enter a file name: ");
gets(filename);

if ( ( fp = fopen ( filename, "w" ) ) != NULL )
{
printf("(\nSucsess\n");

for ( count = 0; count < 5; count++)
{
fprintf( fp, "data[%d] = %f\n", count, data[count] );
fprintf( stdout, "data[%d] = %f", count, data[count] );
}
getch();
}

else
{
puts("Error opening file");
return 0;
}

fclose( fp );

if ( ( fp = fopen ( filename, "r" ) ) != NULL )
{
printf("(\nSucsess\n");

for ( count = 0; count < 5; count++)
{
fscanf( fp,"%f", &data[count] );
fprintf( stdout, "data[%d] = %f\n", count, data[count] );
}

getch();
}

else
{
puts("Error opening file");
return 0;
}

fclose( fp );

return 0;
}
}