Added script for Problem 2