Complete task 2
Fix the script