optmised main and step 1 2 and 4

udpate main file and optmised first and econd step

add all files