added output to scripts
cleaned up assignment code from TA