put an elif in the logic