added bonus 1 - but implemented also else