Updated 5th challenge with real code