Add vote logic