Edit Button fix. Justin Canas