Función que devuelva un mensaje si el número empleados de un departamento es igual o mayor de 3

CREATE FUNCTION check_employees(depart_id INT)
RETURNS varchar(50)
DETERMINISTIC
BEGIN
DECLARE msg VARCHAR(50);
SET msg = 0;
IF (SELECT COUNT(*) FROM employees WHERE department_id = depart_id) >= 3 THEN
SET msg = 'Number of employees is greater than or equal to 3';
END IF;
RETURN (msg);
END;