solved problem #4