Minor update to package