Home work 4 - fix