Completed the script

This should work