dodani if statmenti ki bodo kasneje skrajšani

daljša verzija