added edit step functionality