Implement editig feature