Add voting logic