added edit function...still trying to fix complete