Adding logic for a simple election