Got edit working