Fixes challenge4