updates for milestone 4a