Made revisions for milestone 4 update