Edit script to add for Task 2