Copied code into milestone 2 folder