edit functionality working now