Функция копирования в символьный массив

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define LIM 1000
 
void strng_cpy (char *a, char *b, int count);
 
int main(void)
{
    int max;
    int count;
    char c;
    printf("Введите максимальное количество элементов в строке\n");
    scanf("%d", &max);
    char *string1 = (char *)malloc(max * sizeof(char));
    char *string2 = (char *)malloc(max * sizeof(char));
    do
    {   
        printf("Введите вторую  строку\n");
        
        fgets(string2, max, stdin);
        printf("вы ввели:\n%s\n%s\n", string1, string2);
        printf("Выберите количество символов, которое необходимо скопировать из второй строки\n");
        scanf("%d", &count);
        strng_cpy(string1, string2, count);
        printf("%s\n%s\n", string1, string2);
        printf("Нажмите любую кнопку для продолжения или q для выхода\n");
    }
    while((c = getchar()) != 'q');
    printf("Всего хорошего\n");
}
 
void strng_cpy (char *a, char *b, int count)
{   
    int i = 0;
    while(i++ < count)
    {
        *a++ = *b++;
    }
}

char * strng_cpy (char *a, char *b, int count)
{   
    int i = 0;
    while(i++ < count)
    {
        *a++ = *b++;
    }
return (char *) a;
}

return a

do
    {   
        printf("Введите вторую  строку\n");
        
        fgets(string2, max, stdin);
        printf("вы ввели:\n%s\n%s\n", string1, string2);
        printf("Выберите количество символов, которое необходимо скопировать из второй строки\n");
        scanf("%d", &count);
        
        printf("%s\n%s\n", strng_cpy(string1, string2, count), string2);
        printf("Нажмите любую кнопку для продолжения или q для выхода\n");
        while(getchar()!='\n')
        {
            continue;
        }
    }
    while(getchar() != 'q');
    printf("Всего хорошего\n");
}
 
char * strng_cpy (char *a, char *b, int count)
{   
    int i = 0;
    char *pt;
    while(i++ < count)
    {
        *a++ = *b++;
    }
    *a = 0;
return a;
}