steps towards fixing
Fixed edit bug