Made count actually work