small changes on task 3