updates to assignment #1