added edit fixing issue 2