Minor changes to package