small tweaks
Only add files that are needed