Continue adding edit funcionality