working code for up to mileston5B
working for milestone 5-b