minor changes for wk 2