Fixed the edit function!!! :D