Fix for task4