almost there for milestone 2