problem 6 update
changed stuff