Added edit stuff