Forgot to return count... Fixed!