updates for milestone 4