Updating script from class018_homework