prompt for milestone 3 and refactor