Starting edit function, might need to rename class