added more step functionality