Made task 4 better