Added back lines used for contribution count

A couple lines were removed which broke the contribution count feature. They have been added back.