Ändrat lite logik