fix some bug in count for and start to deal with listcomp