Mistake with int count... fixed