add counting with neurons layer