Lagt till if-satser.