Task4 problem fixed