adding edited code