Fixing task 5