Trzeba by albo w miejsce $count wstawić return albo w miejsce return dołozyć count.