Adding in vote count code