Edit function was added