reg_count.cpp modified