New function

Added edit function, forgot to include in preivous update.