instead of else use elif