update the scripts for assignment 1