Add editing function for session